/*******************************************************************/
/* @ All Rights Reserved Copyright (C) 2014 broada,Co.,Ltd.        */
/*                                                                 */
/*  NAME      = dcom_request.c                                   */
/*  NOTE      =                                                    */
/*  DATE      = 2015/12/18 by geyh                                 */
/*******************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <CUnit/Basic.h>
#include <CUnit/CUnit.h>

#include <dcom/dcom_comm.h>
#include <dcom/dcom_core.h>
#include "frcomm.h"
#include "frmem.h"
#include "conntrack.h"
#include "frct_comm.h"
#include "frbyte_buffer.h"
#include "frlist.h"
#include "frreport_comm.h"
#include "apmtest_cunit.h"
#include "apmtest_comm.h"

#include "../../../src/module/dcom/config.h"

static int __suite_init(void)
{

    return 0;
}

static int __suite_cleanup(void)
{

    return 0;
}

extern L7session_dcom_private* __dcom_private_create(void);
extern int dcom_request(Frbyte_Packet *p_st_data, L7session_dcom_private *p_st_private, Frreport_body_data *p_st_body_data, int i_flag_need);

static void __case1(void)
{
    Frbyte_Packet *p_st_data = NULL;
    Frreport_body_data *p_st_body_data = NULL;
    L7session_dcom_private *p_st_private = NULL;
    int i_ret = 0;
    Conntrack *p_st_conn = NULL;

    p_st_private = __dcom_private_create();
    p_st_private->i_packet_endian = BYTE_BIG_ENDIAN;
    p_st_private->i_packet_type = 0x0100;

    i_ret = apmtest_create_ct(&p_st_conn, dcom_session_destroy, (char *)"192.168.0.1", (char *)"192.168.0.35", 5460, 5672);
    p_st_body_data = frreport_body_new_l7 (p_st_conn);


    //测试修改部分
    p_st_data = bytebuff_creat();
    bytebuff_add(p_st_data,(char *)"\x05\x00\x00\x83\x10\x00\x00\x00\x5c\x00\x00\x00\x4e\x00\x00\x00\x34\x00\x00\x00\x03"
            "\x00\x07\x00\x02\xf8\x00\x00\x08\x11\x00\x00\xd4\x95\x49\x18\x36\x6a\x8e\x6d\x05\x00\x07\x00\x00\x00\x00\x00"
            "\x00\x00\x00\x00\x19\x5c\x6d\x93\x78\x2f\x18\x43\x88\xb2\x27\x0d\xfb\x87\x2b\xeb\x00\x00\x00\x00\x8a\x00\x00"
            "\x00\x75\x00\x00\x00\x8a\x00\x00\x00\x7a\x00\x00\x00\xff\x00\x00\x00", 92);
    bytebuff_skip (p_st_data, 3);

    i_ret = dcom_request(p_st_data, p_st_private, p_st_body_data, FR_YES);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    bytebuff_free(&p_st_data);
    frreport_body_free(&p_st_body_data);
    dcom_session_destroy(p_st_private);
    apmtest_destroy_ct(&p_st_conn);

    return;


}

static void __case2(void)
{
    Frbyte_Packet *p_st_data = NULL;
    Frreport_body_data *p_st_body_data = NULL;
    L7session_dcom_private *p_st_private = NULL;
    int i_ret = 0;
    Conntrack *p_st_conn = NULL;

    p_st_private = __dcom_private_create();
    p_st_private->i_packet_endian = BYTE_BIG_ENDIAN;
    p_st_private->i_packet_type = 0x0100;

    i_ret = apmtest_create_ct(&p_st_conn, dcom_session_destroy, (char *)"192.168.0.1", (char *)"192.168.0.35", 5460, 5672);
    p_st_body_data = frreport_body_new_l7 (p_st_conn);

    //测试修改部分
    p_st_data = bytebuff_creat();
    bytebuff_add(p_st_data,(char *)"\x05\x00\x00\x83\x10\x00\x00\x00\x5c\x00\x00\x00\x4e\x00\x00\x00\x34\x00\x00\x00\x03"
            "\x00\x07\x00\x02\xf8\x00\x00\x08\x11\x00\x00\xd4\x95\x49\x18\x36\x6a\x8e\x6d\x05\x00\x07\x00\x00\x00\x00\x00"
            "\x00\x00\x00\x00\x19\x5c\x6d\x93\x78\x2f\x18\x43\x88\xb2\x27\x0d\xfb\x87\x2b\xeb\x00\x00\x00\x00\x8a\x00\x00"
            "\x00\x75\x00\x00\x00\x8a\x00\x00\x00\x7a\x00\x00\x00\xff\x00\x00\x00", 92);
    bytebuff_skip (p_st_data, 3);

    i_ret = dcom_request(p_st_data, p_st_private, NULL, FR_NO);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_SUCC);

    bytebuff_free(&p_st_data);
    frreport_body_free(&p_st_body_data);
    dcom_session_destroy(p_st_private);
    apmtest_destroy_ct(&p_st_conn);
    return;


}


static void __case3(void)
{
    Frbyte_Packet *p_st_data = NULL;
    Frreport_body_data *p_st_body_data = NULL;
    L7session_dcom_private *p_st_private = NULL;
    int i_ret = 0;
    Conntrack *p_st_conn = NULL;

    p_st_private = __dcom_private_create();
    p_st_private->i_packet_endian = BYTE_BIG_ENDIAN;
    p_st_private->i_packet_type = 0x0100;

    i_ret = apmtest_create_ct(&p_st_conn, dcom_session_destroy, (char *)"192.168.0.1", (char *)"192.168.0.35", 5460, 5672);
    p_st_body_data = frreport_body_new_l7 (p_st_conn);

    //测试修改部分
    p_st_data = bytebuff_creat();
    bytebuff_add(p_st_data,(char *)"\x05\x00\x00\x83\x10\x00\x00\x00\x5c\x00\x00\x00\x4e\x00\x00\x00\x34\x00\x00\x00\x03"
            "\x00\x07\x00\x02\xf8\x00\x00\x08\x11\x00\x00\xd4\x95\x49\x18\x36\x6a\x8e\x6d\x05\x00\x07\x00\x00\x00\x00\x00"
            "\x00\x00\x00\x00\x19\x5c\x6d\x93\x78\x2f\x18\x43\x88\xb2\x27\x0d\xfb\x87\x2b\xeb\x00\x00\x00\x00\x8a\x00\x00"
            "\x00\x75\x00\x00\x00\x8a\x00\x00\x00\x7a\x00\x00\x00\xff\x00\x00\x00", 92);
    bytebuff_skip (p_st_data, 3);

    i_ret = dcom_request(p_st_data, p_st_private, NULL, FR_YES);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_FAIL);

    bytebuff_free(&p_st_data);
    frreport_body_free(&p_st_body_data);
    dcom_session_destroy(p_st_private);
    apmtest_destroy_ct(&p_st_conn);
    return;


}

static void __case4(void)
{
    Frbyte_Packet *p_st_data = NULL;
    Frreport_body_data *p_st_body_data = NULL;
    L7session_dcom_private *p_st_private = NULL;
    int i_ret = 0;
    Conntrack *p_st_conn = NULL;

    p_st_private = __dcom_private_create();
    p_st_private->i_packet_endian = BYTE_BIG_ENDIAN;
    p_st_private->i_packet_type = 0x0100;

    i_ret = apmtest_create_ct(&p_st_conn, dcom_session_destroy, (char *)"192.168.0.1", (char *)"192.168.0.35", 5460, 5672);
    p_st_body_data = frreport_body_new_l7 (p_st_conn);

    //测试修改部分
    p_st_data = bytebuff_creat();
    bytebuff_add(p_st_data,(char *)"\x05\x00\x00\x83\x10\x00\x00\x00\x5c\x00\x00\x00\x4e\x00\x00\x00\x34\x00\x00\x00\x03"
            "\x00\x07\x00\x02\xf8\x00\x00\x08\x11\x00\x00\xd4\x95\x49\x18\x36\x6a\x8e\x6d\x05\x00\x07\x00\x00\x00\x00\x00"
            "\x00\x00\x00\x00\x19\x5c\x6d\x93\x78\x2f\x18\x43\x88\xb2\x27\x0d\xfb\x87\x2b\xeb\x00\x00\x00\x00\x8a\x00\x00"
            "\x00\x75\x00\x00\x00\x8a\x00\x00\x00\x7a\x00\x00\x00\xff\x00\x00\x00", 92);
    bytebuff_skip (p_st_data, 3);


    i_ret = dcom_request(NULL, p_st_private, p_st_body_data, FR_NO);

    CU_ASSERT_EQUAL(i_ret, APM_L7_ANALYZE_FAIL);

    bytebuff_free(&p_st_data);
    frreport_body_free(&p_st_body_data);
    dcom_session_destroy(p_st_private);
    apmtest_destroy_ct(&p_st_conn);
    return;


}


int add_suite_dcom_request(void)
{
    CU_pSuite pSuite = NULL;

    if ((pSuite = CU_add_suite(MODULE_NAME" dcom_request()", __suite_init, __suite_cleanup)) == NULL)
        return FR_FAIL;

    CU_add_test(pSuite, "case1", __case1);
    CU_add_test(pSuite, "case2", __case2);
    CU_add_test(pSuite, "case3", __case3);
    CU_add_test(pSuite, "case4", __case4);

    return FR_SUCC;

}


